package com.example.testClass.easyExcelDemo;

import com.alibaba.excel.EasyExcel;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;

/**
 * @author: tobed
 * @date: 2022/9/15
 * @time: 17:14
 */
@RestController
@RequestMapping("excel")
public class EasyExcelController {

    @PostMapping("exportExcel")
    public void ExportExcel() {
        User user = new User();
        user.setId("1");
        user.setYhmm("用户名密码");
        user.setYhxb("原核细胞");
        user.setYhzh("用户账号");
        user.setYhxm("张三");
        ArrayList<User> list = new ArrayList<>();
        list.add(user);
        System.out.println("我进来了");
        String fileName = TestFileUtil.getPath() + "User" + System.currentTimeMillis() + ".xls";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为用户表 然后文件流会自动关闭
        EasyExcel.write(fileName, User.class).sheet("用户表").doWrite(data());
    }

    public List<User> data() {
        //查询用户表,具体service层实现就不写了
        User user = new User();
        user.setId("1");
        user.setYhmm("用户名密码");
        user.setYhxb("原核细胞");
        user.setYhzh("用户账号");
        user.setYhxm("张三");
        ArrayList<User> list = new ArrayList<>();
        list.add(user);
        return list;
    }

}
